import torch

x = torch.rand(5, 3)
print(x)

y = x.view(15)
z = x.view(-1, 5)  # -1所指的维度可以根据其他维度的值推出来
print(x.size(), y.size(), z.size())
print(y)
print(z)
